Where Post-Training Quantization Breaks Text Embedders: A Measured Map Across Four Embedder Families
文本嵌入模型的后训练量化失效图谱:四个模型家族的实测研究
🧭 模型量化与检索嵌入压缩
ThakiCloud
在检索嵌入模型上系统验证量化启发式建议,发现嵌入表保护和模块敏感度排序均不可迁移,蒸馏学生模型在极端量化下更优
📌 工作参考价值(4 点)
- 量化检索 embedding 模型时不要直接套用 LLM 量化经验:embedding table 并不总是需要最高优先保护,INT4/g16 下各模块独立量化损失差异不足 1 个 NDCG 点,均匀量化即可,无需复杂的混合精度分配。
- 在 INT3 及以下位宽,模块敏感度排序因模型家族而异(Qwen 中 FFN 最敏感,BGE-M3 和 E5 中 attention 最敏感),需要针对自有模型重新测量,不能依赖通用排序。
- 重建误差代理可用于筛选全局位宽,但不可用于选择具体保护哪些 tensor,因为其跨位宽的强预测力是 range-extension 伪影;实际选型应在目标检索任务上做 per-query bootstrap 检验。
- 若需 INT3/INT2 极端压缩,可考虑训练一个更小的蒸馏学生模型并量化,可能比直接量化大模型更优(文中 109M 学生 INT3 在 68.4 MB 下 NDCG@10 达 78.04,优于 0.6B 教师量化版本的 64.46 / 297.9 MB),但注意这种优势仅限蒸馏任务内。